from django.contrib import admin
from lottery.models import CollectionInfoModels, LotteryTheme


# Register your models here.


class CollectionInfoModelsAdmin(admin.ModelAdmin):
    list_display = (
        'id',
        'full_name',
        'email',
        'telegram_username',
        'bep_wallet_address',
        'is_hook',
        'theme',
    )


class LotteryThemeInfoModelsAdmin(admin.ModelAdmin):
    list_display = (
        'id',
        'lottery_theme_name',
        'is_now',
        'draw_at',
        'people_number',
        'contract_address',
    )


admin.site.register(CollectionInfoModels, CollectionInfoModelsAdmin)
admin.site.register(LotteryTheme, LotteryThemeInfoModelsAdmin)

admin.site.site_title = '抽奖平台后端管理系统'
admin.site.site_header = '抽奖平台后端管理系统'
